package com.iscas.loadstorm.isloadstorm.util;

import com.iscas.loadstorm.isloadstorm.domain.TestRun;

import java.util.concurrent.ConcurrentHashMap;


public class LoadCommonUtil {
    // 用于存储测试任务状态的 Map
    private static final ConcurrentHashMap<String, TestRun> testRuns = new ConcurrentHashMap<>();

    /**
     * 查看测试任务状态
     *
     * @param testId 测试任务的唯一 ID
     * @return 测试任务状态
     */
    public static String getTestStatus(String testId) {
        TestRun testRun = testRuns.get(testId);
        if (testRun == null) {
            return "Test ID not found.";
        }

        // 返回包含样本数和错误数的状态信息
        return String.format("Test ID: %s, Samples: %d, Errors: %d",
                testId,
                testRun.getSamples(),
                testRun.getErrors()
        );
    }

    /**
     * 添加测试任务
     *
     * @param testId 测试任务的唯一 ID
     * @param testRun 测试任务对象
     */
    public static void addTestRun(String testId, TestRun testRun) {
        testRuns.put(testId, testRun);
    }
}
